package com.example.demo.controller;

import com.auth0.jwt.JWT;
import com.example.demo.entity.VO.LoginUser;
import com.example.demo.entity.User;
import com.example.demo.service.UserService;
import com.example.demo.utils.JWTUtils;
import com.example.demo.utils.ResponseResult;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.Map;

@RestController
public class UserController {
    @Autowired
    UserService userService;
    @Autowired
    RedisTemplate<String,Object> redisTemplate;
    @PostMapping("/login")
    public ResponseResult<String> login(@RequestBody LoginUser loginUser){
        String storeCode=(String) redisTemplate.opsForValue().get("captcha_codes:"+loginUser.getUuid());
        System.out.println(storeCode+loginUser.toString());
        if (loginUser.getCode().equals(storeCode)){
            User u=userService.getUser(loginUser);
            if(u!=null){
                Map<String,String> map=new HashMap<>();
                map.put("userid",u.getUserId()+"");
                map.put("username",u.getUserName());
                String token= JWTUtils.getToken(map);
                redisTemplate.delete(storeCode);
                return ResponseResult.createBySuccess(200,token,"登录成功");
            }
            return ResponseResult.createBySuccess(500,"","账号或密码错误");
        }else {
            return ResponseResult.createBySuccess(500,"","验证码错误");
        }
    }
    @GetMapping("/getInfo")
    public ResponseResult<User> getInfo(HttpServletRequest request){
        String token =request.getHeader("Authorization");
        System.out.println(token+" ");
        String id= JWT.decode(token.substring(token.lastIndexOf(" ")+1)).getClaim("userid").asString();
        User user=userService.getUserById(Integer.parseInt(id));
        return ResponseResult.createBySuccess(200,user,"查询成功");
    }
}
